#!/usr/bin/env python
#
# Copyright (C) 2011 Clement Lorteau <clem@lorteau.fr>	
#		
# This file is part of QuickScreenShots.		
# 	
# This program is free software; you can redistribute it and/or modify	
# it under the terms of the GNU General Public License as published by		
# the Free Software Foundation; version 3.	
# 	
# This program is distributed in the hope that it will be useful,		
# but WITHOUT ANY WARRANTY; without even the implied warranty of		
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the		
# GNU General Public License for more details.		
# 		
# You should have received a copy of the GNU General Public License		
# along with this program; if not, write to the Free Software		
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.

from wx import Brush
from PIL import Image

def rotate(im, angle=20, background='#FFFFFF'):
    b = Brush(background)
    background = b.GetColour().GetRGB()
    bg = Image.new('RGB', im.rotate(angle, expand=1).size, background)
    im = im.convert('RGBA').rotate(angle, Image.BILINEAR, expand=1)
    bg.paste(im, im)
    return bg
    
if __name__ == "__main__":
    import time
    import windowscreenshot
    time.sleep(3)
    im = windowscreenshot.snap()
    im = rotate(im)
    im.save('screenshot.jpg', 'JPEG')